#!/bin/bash

# 消融实验：基于注意力的探头选择
# 实验ID: Probes-01

# 切换到项目根目录
cd "$(dirname "$0")/.."

# 运行实验，开启基于注意力的探头选择
python run_main_channel.py \
  --model_comment "ablation_attention_probe" \
  --train_epochs 100 \
  --batch_size 4 \
  --learning_rate 0.0005 \
  --islora \
  --lora_r 8 \
  --load_spatial_corr \
  --optim adam \
  --use_attention_probe_selection
  
echo "Probes-01实验完成: 使用基于注意力的探头选择机制" 